from argparse import Namespace
import torch
from pytorch_lightning.loggers import WandbLogger
import time



config = Namespace(
    project_name = 'mnist原始',
    AVAIL_GPUS = min(1, torch.cuda.device_count()),
    BATCH_SIZE = 256 if min(1, torch.cuda.device_count()) else 64,
    lr = 0.02,
    optim_type = "Adam",
    max_epochs = 3,
    ckpt_path = "best.ckpt",
    data_dir ="../../../../data/",
    dropout_p = 0.1,
    modelPATH = "modelPATH"
)


